package me.timlong;

import java.util.Arrays;

public class CandyDistribution {

    /**
     * 题目描述
     * 有N个小朋友站在一排，每个小朋友都有一个评分
     * 你现在要按以下的规则给孩子们分糖果：
     * 每个小朋友至少要分得一颗糖果
     * 分数高的小朋友要他比旁边得分低的小朋友分得的糖果多
     * 你最少要分发多少颗糖果？
     * @param ratings
     * @return
     */
    public int candy(int[] ratings) {
        int len = ratings.length;
        int[] candies = new int[len + 1];
        int sumCandies = 0;

        // 初始化，先给每个小朋友发一个糖果
        Arrays.fill(candies, 1);
        // 从左向右，保证这个方向上每个比右手边得分高的小朋友得到糖数更多
        for (int i = 0; i < len - 1; i++) {
            if(ratings[i + 1] > ratings[i])
                candies[i + 1] = Math.max(candies[i + 1], candies[i] + 1);
        }

        // 从右向左，保证这个方向上每个比左手边得分高的小朋友得到糖数更多
        for (int i = len - 2; i >= 0 ; i--) {
            if(ratings[i] > ratings[i + 1])
                candies[i] = Math.max(candies[i], candies[i + 1] + 1);
        }


        for (int i = 0; i < len; i++)
            sumCandies += candies[i];

        return sumCandies;
    }

    public static void main(String[] args) {
//        2, 1, 3, 6, 2, 4, 7, 5, 6, 8, 1, 2, 5
        int[] arr = {2, 3, 4};
        System.out.println(new CandyDistribution().candy(arr));


    }

}
